fix for model loading