Fix devise bugs