Fix child model being shown in UI