config/routes.rb in switch_user-0.9.1 vs config/routes.rb in switch_user-0.9.2

@@ -1,5 +1,12 @@
+# TODO can this be writting once so that it is rails3 and rails4 compatible?
 if Rails.version =~ /^3/
   Rails.application.routes.draw do
     match 'switch_user' => 'switch_user#set_current_user'
+    match 'switch_user/remember_user' => 'switch_user#remember_user'
   end
-end
\ No newline at end of file
+elsif Rails.version =~ /^4/
+  Rails.application.routes.draw do
+    get :switch_user, :to => 'switch_user#set_current_user'
+    get 'switch_user/remember_user', :to => 'switch_user#remember_user'
+  end
+end