Fixes model loading