Fix ng-model and clean up test