diff --git a/invokeai/backend/flux/extensions/kontext_extension.py b/invokeai/backend/flux/extensions/kontext_extension.py
index e7a852775d..8f22189e76 100644
--- a/invokeai/backend/flux/extensions/kontext_extension.py
+++ b/invokeai/backend/flux/extensions/kontext_extension.py
@@ -1,10 +1,11 @@
+from dataclasses import dataclass
+
 import einops
 import numpy as np
 import torch
+import torchvision.transforms as T
 from einops import repeat
 from PIL import Image
-import torchvision.transforms as T
-from dataclasses import dataclass
 
 from invokeai.app.invocations.fields import FluxKontextConditioningField
 from invokeai.app.invocations.flux_vae_encode import FluxVaeEncodeInvocation