diff --git a/dpdata/pymatgen/structure.py b/dpdata/pymatgen/structure.py
index fa709963..b6c148da 100644
--- a/dpdata/pymatgen/structure.py
+++ b/dpdata/pymatgen/structure.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 try:
-    from pymatgen.core import Structure # noqa: F401
+    from pymatgen.core import Structure  # noqa: F401
 except ImportError:
     pass
 
@@ -19,6 +19,6 @@ def from_system_data(structure) -> dict:
         "atom_numbs": atom_numbs,
         "atom_types": atom_types,
         "coords": np.array([coords]),
-        "cells": np.array([cells])
+        "cells": np.array([cells]),
     }
     return info_dict
diff --git a/tests/test_from_pymatgen.py b/tests/test_from_pymatgen.py
index 0e5740fb..d3ddbe3e 100644
--- a/tests/test_from_pymatgen.py
+++ b/tests/test_from_pymatgen.py
@@ -1,7 +1,7 @@
 import os
 import unittest
 
-from comp_sys import CompSys, IsPBC
+from comp_sys import CompSys
 from context import dpdata
 
 try: