diff --git a/src/VirtualColumn.php b/src/VirtualColumn.php
index 1943a80..82de5aa 100644
--- a/src/VirtualColumn.php
+++ b/src/VirtualColumn.php
@@ -140,9 +140,14 @@ trait VirtualColumn
         ];
     }
 
+    /**
+     * Get a column name for an attribute that can be used in SQL queries.
+     *
+     * (`foo` or `data->foo` depending on whether `foo` is in custom columns)
+     */
     public function getColumnForQuery(string $column): string
     {
-        if (in_array($column, static::getCustomColumns())) {
+        if (in_array($column, static::getCustomColumns(), true)) {
             return $column;
         }