Adjusting to make feature work with new version of Devise?