Fixed : wrong model was being called