diff --git a/Java/src/main/java/org/ciyam/at/API.java b/Java/src/main/java/org/ciyam/at/API.java
index 4b94c35..7244da6 100644
--- a/Java/src/main/java/org/ciyam/at/API.java
+++ b/Java/src/main/java/org/ciyam/at/API.java
@@ -146,7 +146,7 @@ public abstract class API {
 		state.rewindCodePosition(offset);
 	}
 
-	protected void setSleepUntilHeight(MachineState state, int height) {
+	protected void setSleepUntilHeight(MachineState state, Integer height) {
 		state.setSleepUntilHeight(height);
 	}