diff --git a/mdagent/tools/base_tools/preprocess_tools/packing.py b/mdagent/tools/base_tools/preprocess_tools/packing.py
index 9e5843f4..d067a138 100644
--- a/mdagent/tools/base_tools/preprocess_tools/packing.py
+++ b/mdagent/tools/base_tools/preprocess_tools/packing.py
@@ -205,7 +205,6 @@ def packmol_wrapper(
 
 
 """Args schema for packmol_wrapper tool. Useful for OpenAI functions"""
-##TODO
 
 
 class PackmolInput(BaseModel):
diff --git a/tests/test_packing.py b/tests/test_packing.py
index cfe96865..77b4c47d 100644
--- a/tests/test_packing.py
+++ b/tests/test_packing.py
@@ -2,7 +2,11 @@
 
 import pytest
 
-from mdagent.tools.base_tools.preprocess_tools.packing import Molecule, PackmolBox
+from mdagent.tools.base_tools.preprocess_tools.packing import (
+    Molecule,
+    PackmolBox,
+    PackMolTool,
+)
 from mdagent.utils import PathRegistry
 
 
@@ -16,6 +20,11 @@ def packmolbox(get_registry):
     return PackmolBox(get_registry)
 
 
+@pytest.fixture
+def packmoltool(get_registry):
+    return PackMolTool(get_registry)
+
+
 @pytest.fixture
 def dummy_molecule():
     return Molecule(