diff --git a/build/lib/pylfsr/pylfsr.py b/build/lib/pylfsr/pylfsr.py
index 9c84a4f..bff6b6a 100644
--- a/build/lib/pylfsr/pylfsr.py
+++ b/build/lib/pylfsr/pylfsr.py
@@ -543,7 +543,7 @@ def get_Ifpoly(self,fpoly):
 		ifpoly: polynomial as list e.g. [5,3] for x^5 + x^3 + 1
 
 		'''
-		if isinstance(fpoly, list) or (isinstance(fpoly, numpy.ndarray) and len(fpoly.shape)==1):
+		if isinstance(fpoly, list) or (isinstance(fpoly, np.ndarray) and len(fpoly.shape)==1):
 			fpoly = list(fpoly)
 			fpoly.sort(reverse=True)
 			ifpoly = [fpoly[0]] +[fpoly[0]-ff for ff in fpoly[1:]]
diff --git a/pylfsr/pylfsr.py b/pylfsr/pylfsr.py
index b8483cd..f7d0b37 100644
--- a/pylfsr/pylfsr.py
+++ b/pylfsr/pylfsr.py
@@ -556,7 +556,7 @@ def get_Ifpoly(self,fpoly):
 		ifpoly: polynomial as list e.g. [5,3] for x^5 + x^3 + 1
 
 		'''
-		if isinstance(fpoly, list) or (isinstance(fpoly, numpy.ndarray) and len(fpoly.shape)==1):
+		if isinstance(fpoly, list) or (isinstance(fpoly, np.ndarray) and len(fpoly.shape)==1):
 			fpoly = list(fpoly)
 			fpoly.sort(reverse=True)
 			ifpoly = [fpoly[0]] +[fpoly[0]-ff for ff in fpoly[1:]]