Update src/systems/unit_check.jl

Author: ChrisRackauckas

src/systems/unit_check.jl

@@ -78,9 +78,7 @@ get_unit(op::typeof(instream), args) = get_unit(args[1])
 function get_unit(op, args) # Fallback
     result = oneunit(op(get_unit.(args)...))
     try
-        result = op(unit_args...)
-        # For operations that return a unit directly, return oneunit to get the unit structure
-        return oneunit(result)
+        get_unit(result)
     catch
         try
             # Try with oneunit for numeric operations