Fixing broken test case

Need to add ng-model to the directive otherwise it causes an error.

@donnapep @stulees please review & merge. Thanks.