diff --git a/source/tests/common/dpmodel/test_from_ijs.py b/source/tests/common/dpmodel/test_from_ijs.py
index dc0af94037..adce99c5e5 100644
--- a/source/tests/common/dpmodel/test_from_ijs.py
+++ b/source/tests/common/dpmodel/test_from_ijs.py
@@ -85,10 +85,6 @@ def test_ase_matches_intree_carry_all_nonperiodic(self) -> None:
         ng_ref = build_neighbor_graph(coord, atype, None, rcut=4.0)
         self.assertEqual(self._sets(ng_ase, 6), self._sets(ng_ref, 6))
 
-
-if __name__ == "__main__":
-    unittest.main()
-
     def test_ase_excludes_virtual_atoms_like_dense(self) -> None:
         """Virtual atoms (atype < 0) excluded as center AND neighbor.
 
@@ -113,3 +109,7 @@ def test_ase_excludes_virtual_atoms_like_dense(self) -> None:
         ei = ng_ase.edge_index[:, ng_ase.edge_mask]
         flat_atype = atype.reshape(-1)
         assert np.all(flat_atype[ei[0]] >= 0) and np.all(flat_atype[ei[1]] >= 0)
+
+
+if __name__ == "__main__":
+    unittest.main()