Fix for model