diff --git a/test/jaxmd.py b/test/jaxmd.py
index bfce2369..ba3105ec 100644
--- a/test/jaxmd.py
+++ b/test/jaxmd.py
@@ -256,6 +256,8 @@ def forward(
 
 if __name__ == "__main__":
     import platform
+
     # Deps not available on macos
-    if platform.system() == 'Darwin': return
+    if platform.system() == "Darwin":
+        return
     absltest.main()