Fix issue with devise registration