diff --git a/app/controllers/application_controller.rb b/app/controllers/application_controller.rb
index 9f0b154cbe261238a85a8f26962191f7d83281a7..7c9380a05eb378c2c355c5307d5bde9ffbb591ed 100644
--- a/app/controllers/application_controller.rb
+++ b/app/controllers/application_controller.rb
@@ -23,13 +23,13 @@ class ApplicationController < ActionController::API
     registration_params = [:name, :email, :avatar, :password, :password_confirmation]
 
     if params[:action] == 'update'
-      devise_parameter_sanitizer.for(:account_update) {
-          |u| u.permit(registration_params << :current_password)
-      }
+      devise_parameter_sanitizer.permit(:account_update) do |user_params|
+        user_params.permit(registration_params << :current_password)
+      end
     elsif params[:action] == 'create'
-      devise_parameter_sanitizer.for(:sign_up) {
-          |u| u.permit(registration_params << :terms_of_service)
-      }
+      devise_parameter_sanitizer.permit(:sign_in) do |user_params|
+        user_params.permit(registration_params << :terms_of_service)
+      end
     end
   end
 
@@ -38,5 +38,4 @@ class ApplicationController < ActionController::API
   def user_not_authorized
     render nothing: true, status: :unauthorized
   end
-
 end