diff --git a/src/lib/Gateway/ExpressionVisitor.php b/src/lib/Gateway/ExpressionVisitor.php
index 3bd5bd0..2492f15 100644
--- a/src/lib/Gateway/ExpressionVisitor.php
+++ b/src/lib/Gateway/ExpressionVisitor.php
@@ -115,17 +115,17 @@ public function walkComparison(Comparison $comparison)
             $relationshipType = get_class($relationship);
 
             switch ($relationshipType) {
-                case DoctrineOneToManyRelationship::class:
-                case PreJoinedDoctrineRelationship::class:
-                    return $this->handleOneToManyRelationship(
+                case DoctrineRelationship::class:
+                    return $this->handleRelationship(
                         $relationshipMetadata,
+                        $relationship->getForeignKeyColumn(),
                         $foreignClassProperty,
                         $comparison->getValue(),
                     );
-                case DoctrineRelationship::class:
-                    return $this->handleRelationship(
+                case DoctrineOneToManyRelationship::class:
+                case PreJoinedDoctrineRelationship::class:
+                    return $this->handleOneToManyRelationship(
                         $relationshipMetadata,
-                        $relationship->getForeignKeyColumn(),
                         $foreignClassProperty,
                         $comparison->getValue(),
                     );