AngularJS controller fix