diff --git a/app/controllers/default/units_controller.rb b/app/controllers/default/units_controller.rb
index e65b8fa..d117005 100644
--- a/app/controllers/default/units_controller.rb
+++ b/app/controllers/default/units_controller.rb
@@ -40,6 +40,6 @@ class Default::UnitsController < ApplicationController
 
   def find_unit(user)
     @unit = Unit.find_by!(id: params[:id], user: user)
-    @base = Unit.find_by!(symbol: @unit.base.symbol, user: user? ? nil : user) if @unit.base
+    @base = Unit.find_by!(symbol: @unit.base.symbol, user: user ? nil : current_user) if @unit.base
   end
 end