diff --git a/m3gnet/graph/_converters.py b/m3gnet/graph/_converters.py
index ac5afee..06594c7 100644
--- a/m3gnet/graph/_converters.py
+++ b/m3gnet/graph/_converters.py
@@ -78,7 +78,7 @@ def get_states(self, structure: StructureOrMolecule):
         return states
 
     @abstractmethod
-    def convert(self, structure: StructureOrMolecule, **kwargs) -> MaterialGraph:
+    def convert(self, structure: StructureOrMolecule, state_attr=None, *args, **kwargs) -> MaterialGraph:
         """
         Convert the structure into a graph
         Args: