Fixing model loading script