Updating: fixing the angular controller