Fix integration with angular-ui-router