diff --git a/abstract-machine/am/src/platform/nemu/ioe/gpu.c b/abstract-machine/am/src/platform/nemu/ioe/gpu.c
index 59cc0be..7f213b6 100644
--- a/abstract-machine/am/src/platform/nemu/ioe/gpu.c
+++ b/abstract-machine/am/src/platform/nemu/ioe/gpu.c
@@ -6,14 +6,14 @@
 #define GET_H(var) uint16_t var = inw(VGACTL_ADDR)
 
 void __am_gpu_init() {
-  // int i;
-  // GET_W(w);
-  // GET_H(h);
-  // // int w = inw(VGACTL_ADDR + 2);  // TODO: get the correct width
-  // // int h = inw(VGACTL_ADDR);  // TODO: get the correct height
-  // uint32_t *fb = (uint32_t *)(uintptr_t)FB_ADDR;
-  // for (i = 0; i < w * h; i ++) fb[i] = i;
-  // outl(SYNC_ADDR, 1);
+  int i;
+  GET_W(w);
+  GET_H(h);
+  // int w = inw(VGACTL_ADDR + 2);  // TODO: get the correct width
+  // int h = inw(VGACTL_ADDR);  // TODO: get the correct height
+  uint32_t *fb = (uint32_t *)(uintptr_t)FB_ADDR;
+  for (i = 0; i < w * h; i ++) fb[i] = i;
+  outl(SYNC_ADDR, 1);
 }
 
 void __am_gpu_config(AM_GPU_CONFIG_T *cfg) {