diff --git a/hamilton/plugins/h_polars.py b/hamilton/plugins/h_polars.py
index c59ec5302..b2f847b86 100644
--- a/hamilton/plugins/h_polars.py
+++ b/hamilton/plugins/h_polars.py
@@ -1,6 +1,7 @@
 from typing import Any, Dict, Type, Union
 
 import polars as pl
+
 from hamilton import base
 
 
diff --git a/setup.cfg b/setup.cfg
index b8abec40d..6b1dae301 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -12,3 +12,6 @@ ignore =
     W503,
     # invalid escape sequence
     W605
+
+[isort]
+known_first_party=hamilton
diff --git a/tests/plugins/test_polars_extensions.py b/tests/plugins/test_polars_extensions.py
index 00e6825f4..473ff9e42 100644
--- a/tests/plugins/test_polars_extensions.py
+++ b/tests/plugins/test_polars_extensions.py
@@ -1,8 +1,8 @@
 import pathlib
 
+import polars as pl
 import pytest
 
-import polars as pl
 from hamilton.plugins.polars_extensions import (
     PolarsAvroReader,
     PolarsAvroWriter,
diff --git a/tests/test_graph.py b/tests/test_graph.py
index 6451c66dc..a8a83a653 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -5,9 +5,6 @@
 
 import pandas as pd
 import pytest
-
-import hamilton.graph_utils
-import hamilton.htypes
 import tests.resources.bad_functions
 import tests.resources.config_modifier
 import tests.resources.cyclic_functions
@@ -22,6 +19,9 @@
 import tests.resources.parametrized_nodes
 import tests.resources.test_default_args
 import tests.resources.typing_vs_not_typing
+
+import hamilton.graph_utils
+import hamilton.htypes
 from hamilton import ad_hoc_utils, base, graph, node
 from hamilton.execution import graph_functions
 from hamilton.node import NodeType