Fixing the model attach