diff --git a/src/wafer_map/gen_fake_data.py b/src/wafer_map/gen_fake_data.py
index d65ccbe..04cc94d 100644
--- a/src/wafer_map/gen_fake_data.py
+++ b/src/wafer_map/gen_fake_data.py
@@ -7,17 +7,11 @@
 import math
 import random
 
-from wafer_map import PY2
 from wafer_map import wm_constants as wm_const
 from wafer_map import wm_info
 from wafer_map import wm_utils
 
 
-# Python2 Compatibility
-if PY2:
-    range = xrange
-
-
 def generate_fake_data(**kwargs):
     """
     Generate fake data for wafer_map.