diff --git a/io.asm b/io.asm
index 11dc5ed..4bd50b9 100644
--- a/io.asm
+++ b/io.asm
@@ -60,8 +60,8 @@ init:
 	; AX = memory size in kilobytes
 	; We want it in paragraphs
 	; There are 64 paragraphs in a kilobyte
-	mov cx, 64
-	shl al, cl
+	mov cx, 6
+	shl ax, cl
 	; Memory size is passed in dx
 	mov dx, ax