Fixing load model from directory