diff --git a/tdgl/solver/solver.py b/tdgl/solver/solver.py
index d855e44..73b0544 100644
--- a/tdgl/solver/solver.py
+++ b/tdgl/solver/solver.py
@@ -249,7 +249,7 @@ def current_func(t):
                 return terminal_currents
 
         J_scale = 4 * ((ureg(current_units) / length_units) / K0).to_base_units()
-        assert "dimensionless" in str(J_scale.units), str(J_scale.units)
+        assert J_scale.dimensionless, str(J_scale)
         J_scale = J_scale.magnitude
         self.current_func = lambda t: {
             key: J_scale * value for key, value in current_func(t).items()
diff --git a/tdgl/version.py b/tdgl/version.py
index f5f4a9b..25223ef 100644
--- a/tdgl/version.py
+++ b/tdgl/version.py
@@ -1,4 +1,4 @@
-__version_info__ = (0, 8, 1)
+__version_info__ = (0, 8, 2)
 __version__ = ".".join(map(str, __version_info__))