Only eagerly check DQ

AayushSabharwal · Jan 3, 2024 · 089649e · 089649e
1 parent fc87dd6
commit 089649e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/systems/unit_check.jl b/src/systems/unit_check.jl
@@ -20,7 +20,7 @@ function __get_literal_unit(x)
     end
     v = value(x)
     u = getmetadata(v, VariableUnit, nothing)
-    u === nothing ? nothing : screen_unit(u)
+    u isa DQ.AbstractQuantity ? screen_unit(u) : u
 end
 function __get_scalar_unit_type(v)
     u = __get_literal_unit(v)