diff --git a/ballista/executor/src/executor.rs b/ballista/executor/src/executor.rs
index 84a3bdfa5..60db9f00b 100644
--- a/ballista/executor/src/executor.rs
+++ b/ballista/executor/src/executor.rs
@@ -210,7 +210,7 @@ mod test {
 
     use crate::execution_engine::DefaultQueryStageExec;
     use ballista_core::serde::scheduler::PartitionId;
-    use datafusion::error::DataFusionError;
+    use datafusion::error::{DataFusionError, Result};
     use datafusion::physical_expr::PhysicalSortExpr;
     use datafusion::physical_plan::{
         DisplayAs, DisplayFormatType, ExecutionPlan, Partitioning, RecordBatchStream,