diff --git a/vhost-device-gpu/src/virtio_gpu.rs b/vhost-device-gpu/src/virtio_gpu.rs
index 198650e..4d362d0 100644
--- a/vhost-device-gpu/src/virtio_gpu.rs
+++ b/vhost-device-gpu/src/virtio_gpu.rs
@@ -18,11 +18,12 @@ use rutabaga_gfx::{
     RutabagaComponentType, RutabagaFence, RutabagaFenceHandler, RutabagaHandle,
     RutabagaIntoRawDescriptor, RutabagaIovec, Transfer3D, RUTABAGA_HANDLE_TYPE_MEM_DMABUF,
 };
+#[cfg(feature = "gfxstream")]
+use vhost::vhost_user::gpu_message::VhostUserGpuScanout;
 use vhost::vhost_user::{
     gpu_message::{
         VhostUserGpuCursorPos, VhostUserGpuCursorUpdate, VhostUserGpuDMABUFScanout,
-        VhostUserGpuDMABUFScanout2, VhostUserGpuEdidRequest, VhostUserGpuScanout,
-        VhostUserGpuUpdate,
+        VhostUserGpuDMABUFScanout2, VhostUserGpuEdidRequest, VhostUserGpuUpdate,
     },
     GpuBackend,
 };