Fix for model manager test