diff --git a/tests/VirtualColumnTest.php b/tests/VirtualColumnTest.php
index c7af2b0..0d5dd6c 100644
--- a/tests/VirtualColumnTest.php
+++ b/tests/VirtualColumnTest.php
@@ -183,6 +183,15 @@ class MyModel extends ParentModel
 
 class FooModel extends ParentModel
 {
+    public function getCustomColumns(): array
+    {
+        return [
+            'id',
+            'custom1',
+            'custom2',
+        ];
+    }
+
     public function getDataColumn(): string
     {
         return 'virtual';