diff --git a/pyproject.toml b/pyproject.toml
index b6fcfd7..945b94b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -43,6 +43,9 @@ extend-exclude = [
 select = ["E", "F", "I", "UP", "B", "C4", "ANN"]
 ignore = ["ANN401"]
 
+[tool.ruff.lint.isort]
+known-first-party = ["simple_syrup"]
+
 [tool.mypy]
 python_version = "3.12"
 warn_return_any = true
diff --git a/tests/test_model_attn1_replacement_mutation.py b/tests/test_model_attn1_replacement_mutation.py
index be5ab4b..5e71730 100644
--- a/tests/test_model_attn1_replacement_mutation.py
+++ b/tests/test_model_attn1_replacement_mutation.py
@@ -11,10 +11,11 @@ from typing import Any
 import pytest
 import torch
 from comfy.model_patcher import ModelPatcher
+from torch import nn
+
 from simple_syrup.runtime.model_attention_replacement_mutations import (
     ModelAttn1ReplacementsMutation,
 )
-from torch import nn
 
 
 def test_attn1_replacements_install_every_declared_key_atomically() -> None: