diff --git a/src/hedron/ui/grid.cr b/src/hedron/ui/grid.cr
index d1704a9..a03cfc1 100644
--- a/src/hedron/ui/grid.cr
+++ b/src/hedron/ui/grid.cr
@@ -10,12 +10,12 @@ module Hedron
     property align_x : Align
     property align_y : Align
 
-    def initialize(@size, @expand, @align_x, @align_y); end
+    def initialize(@size, @expand, @align_x : Align, @align_y : Align); end
   end
 
   class Grid < Control
     def initialize
-      @this = UI.new_grid
+      @this = ui_control(UI.new_grid)
     end
 
     def initialize(@this); end