diff --git a/virtualizationservice/Android.bp b/virtualizationservice/Android.bp
index d0dde42..c6add24 100644
--- a/virtualizationservice/Android.bp
+++ b/virtualizationservice/Android.bp
@@ -82,9 +82,10 @@
 
 rust_test {
     name: "virtualizationservice_device_test",
-    srcs: ["src/main.rs"],
+    srcs: ["src/virtmgr.rs"],
     defaults: ["virtualizationservice_defaults"],
     rustlibs: [
+        "libclap",
         "libtempfile",
     ],
     test_suites: ["general-tests"],