diff --git a/composable_lora_step.py b/composable_lora_step.py
index 504695a..2d01887 100644
--- a/composable_lora_step.py
+++ b/composable_lora_step.py
@@ -1,4 +1,4 @@
-from typing import List, Dict
+from typing import List
 import re
 import json
 import math
@@ -6,7 +6,7 @@ import sys
 import traceback
 import random
 
-from modules import extra_networks, shared
+from modules import extra_networks
 
 re_AND = re.compile(r"\bAND\b")
 
diff --git a/plot_helper.py b/plot_helper.py
index bb36e02..dec4942 100644
--- a/plot_helper.py
+++ b/plot_helper.py
@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Tuple, List, Union
+from typing import List
 import io
 import matplotlib
 matplotlib.use('Agg')