Fix TestSqlStageExecution.testSplitAssignment()