diff --git a/tests/unit/model/grids/test_grid_search.py b/tests/unit/model/grids/test_grid_search.py
index acf40a5..f3ac243 100644
--- a/tests/unit/model/grids/test_grid_search.py
+++ b/tests/unit/model/grids/test_grid_search.py
@@ -13,7 +13,7 @@
 
 def test_grid_get_nearest_substation_node(basic_grid):
     substation_node = basic_grid.get_nearest_substation_node(node_id=103)
-    assert NodeType.SUBSTATION_NODE == NodeType(substation_node.node_type)
+    assert NodeType.SUBSTATION_NODE == substation_node.node_type
 
 
 def test_grid_get_nearest_substation_node_no_substation(basic_grid):