[fix] Model loading