diff --git a/deeplay/blocks/base.py b/deeplay/blocks/base.py
index b85a8dba..cf775d85 100644
--- a/deeplay/blocks/base.py
+++ b/deeplay/blocks/base.py
@@ -1,5 +1,4 @@
 from ast import Attribute
-from re import A, T
 from typing import Any, List, Optional, Type, Union, Tuple
 from abc import ABC, abstractmethod
 from warnings import warn
diff --git a/deeplay/tests/blocks/test_base.py b/deeplay/tests/blocks/test_base.py
index 70ca819f..9c21fcb0 100644
--- a/deeplay/tests/blocks/test_base.py
+++ b/deeplay/tests/blocks/test_base.py
@@ -1,5 +1,4 @@
 from itertools import product
-from re import T
 import unittest
 
 import torch