Fix single model case.