diff --git a/tests/framework/grid/CheckboxColumnTest.php b/tests/framework/grid/CheckboxColumnTest.php
index a97d0e3e79..b69b795087 100644
--- a/tests/framework/grid/CheckboxColumnTest.php
+++ b/tests/framework/grid/CheckboxColumnTest.php
@@ -56,19 +56,20 @@ class CheckboxColumnTest extends TestCase
 
     public function testInputValue()
     {
-        $column = new CheckboxColumn();
+        $column = new CheckboxColumn(['grid' => $this->getGrid()]);
         $this->assertContains('value="1"', $column->renderDataCell([], 1, 0));
         $this->assertContains('value="42"', $column->renderDataCell([], 42, 0));
         $this->assertContains('value="[1,42]"', $column->renderDataCell([], [1, 42], 0));
 
-        $column = new CheckboxColumn(['checkboxOptions' => ['value' => 42]]);
+        $column = new CheckboxColumn(['checkboxOptions' => ['value' => 42], 'grid' => $this->getGrid()]);
         $this->assertNotContains('value="1"', $column->renderDataCell([], 1, 0));
         $this->assertContains('value="42"', $column->renderDataCell([], 1, 0));
 
         $column = new CheckboxColumn([
             'checkboxOptions' => function ($model, $key, $index, $column) {
                 return [];
-            }
+            },
+            'grid' => $this->getGrid()
         ]);
         $this->assertContains('value="1"', $column->renderDataCell([], 1, 0));
         $this->assertContains('value="42"', $column->renderDataCell([], 42, 0));
@@ -77,7 +78,8 @@ class CheckboxColumnTest extends TestCase
         $column = new CheckboxColumn([
             'checkboxOptions' => function ($model, $key, $index, $column) {
                 return ['value' => 42];
-            }
+            },
+            'grid' => $this->getGrid()
         ]);
         $this->assertNotContains('value="1"', $column->renderDataCell([], 1, 0));
         $this->assertContains('value="42"', $column->renderDataCell([], 1, 0));