diff --git a/dataset.py b/dataset.py
index 6d242ac..0fb9942 100644
--- a/dataset.py
+++ b/dataset.py
@@ -72,7 +72,6 @@ def generate_dateset(game):
     dataset_env = DatasetEnv(env)
     env = ProcessFrame(dataset_env, 84)
     env = NormalizeFrame(env)
-    env = ClippedRewardsWrapper(env)
 
     ep = 0
     max_ep = 200