diff --git a/vhost-device-gpu/src/virtio_gpu.rs b/vhost-device-gpu/src/virtio_gpu.rs
index 4d362d0..adc8f61 100644
--- a/vhost-device-gpu/src/virtio_gpu.rs
+++ b/vhost-device-gpu/src/virtio_gpu.rs
@@ -1099,11 +1099,13 @@ impl VirtioGpu for RutabagaVirtioGpu {
 #[cfg(test)]
 mod tests {
     use std::{
-        env::set_var,
         os::unix::net::UnixStream,
         sync::{Arc, Mutex},
     };
 
+    #[cfg(feature = "gfxstream")]
+    use std::env::set_var;
+
     use assert_matches::assert_matches;
     use rusty_fork::rusty_fork_test;
     use rutabaga_gfx::{