diff --git a/scala/src/main/scala/execution/PITJoinExec.scala b/scala/src/main/scala/execution/PITJoinExec.scala
index f9423f7..c3f14cc 100644
--- a/scala/src/main/scala/execution/PITJoinExec.scala
+++ b/scala/src/main/scala/execution/PITJoinExec.scala
@@ -72,7 +72,8 @@ protected[pit] case class PITJoinExec(
 
   override def requiredChildDistribution: Seq[Distribution] = {
     if (leftEquiKeys.isEmpty || rightEquiKeys.isEmpty) {
-      UnspecifiedDistribution :: UnspecifiedDistribution :: Nil
+      // TODO: This should be improved, but for now just keep everything in one partition
+      AllTuples :: AllTuples :: Nil
     } else {
       HashClusteredDistribution(leftEquiKeys) :: HashClusteredDistribution(
         rightEquiKeys