Update to handle angular-ui breaking changes

Update to handle angular-ui breaking changes