diff --git a/python/src/lib.rs b/python/src/lib.rs
index bfe089b579..3d159efea0 100644
--- a/python/src/lib.rs
+++ b/python/src/lib.rs
@@ -25,6 +25,7 @@ use raphtory_core::python::{
 };
 use raphtory_storage::python::packages::algorithms::*;
 
+#[cfg(feature = "arrow")]
 use raphtory_core::python::graph::arrow::{PyArrowGraph, PyGraphQuery, PyState};
 
 macro_rules! add_functions {