From 089649e692f03a3754e2e8faccf71d9c51f6b70e Mon Sep 17 00:00:00 2001
From: Yingbo Ma <mayingbo5@gmail.com>
Date: Wed, 3 Jan 2024 14:21:53 -0500
Subject: [PATCH] Only eagerly check DQ

---
 src/systems/unit_check.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/systems/unit_check.jl b/src/systems/unit_check.jl
index 02567fa6a9..9b474fcc1d 100644
--- 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)