revert the model change i had made before

Change-Id: I23ab944762412efa98f74e3662f385b9370431f7