diff --git a/tests/framework/db/QueryBuilderTest.php b/tests/framework/db/QueryBuilderTest.php
index a811248512..df274b16ba 100644
--- a/tests/framework/db/QueryBuilderTest.php
+++ b/tests/framework/db/QueryBuilderTest.php
@@ -959,9 +959,10 @@ abstract class QueryBuilderTest extends DatabaseTestCase
         $i = 0;
         foreach ($this->columnTypes() as $item) {
             list ($column, $builder, $expected) = $item;
-            if (!(strncmp($column, 'pk', 2) === 0 ||
-                  strncmp($column, 'bigpk', 5) === 0 ||
-                  strncmp($column, 'unsignedpk', 10) === 0)) {
+            if (!(strncmp($column, Schema::TYPE_PK, 2) === 0 ||
+                  strncmp($column, Schema::TYPE_UPK, 3) === 0 ||
+                  strncmp($column, Schema::TYPE_BIGPK, 5) === 0 ||
+                  strncmp($column, Schema::TYPE_UBIGPK, 6) === 0)) {
                 $columns['col' . ++$i] = str_replace('CHECK (value', 'CHECK ([[col' . $i . ']]', $column);
             }
         }