diff --git a/tetris/rulesets/thenext.lua b/tetris/rulesets/thenext.lua
index 190e488..0938814 100644
--- a/tetris/rulesets/thenext.lua
+++ b/tetris/rulesets/thenext.lua
@@ -88,12 +88,11 @@ TheNext.wallkicks_ccw = {{x=-1, y=0}, {x=0, y=1}, {x=1, y=0}, {x=0, y=-1}}
 TheNext.wallkicks_cw = {{x=1, y=0}, {x=0, y=1}, {x=-1, y=0}, {x=0, y=-1}}
 
 function TheNext:attemptWallkicks(piece, new_piece, rot_dir, grid)
-
+	
 	local kicks
     if piece.shape == "O" then
 		return
-	elseif new_piece.rotation == piece.rotation + 1 or
-    (piece.rotation == 3 and new_piece.rotation == 0) then
+	elseif rot_dir == 1 then
 		kicks = TheNext.wallkicks_cw
 	else
 		kicks = TheNext.wallkicks_ccw
@@ -127,4 +126,4 @@ end
 
 function TheNext:getDefaultOrientation() return 3 end
 
-return TheNext
+return TheNext
\ No newline at end of file