--- a/test/hotspot/jtreg/sanity/WBApi.java	Wed Jan 17 16:49:43 2018 -0800
+++ b/test/hotspot/jtreg/sanity/WBApi.java	Wed Jan 17 21:44:44 2018 -0800
@@ -27,7 +27,7 @@
  * @modules java.base/jdk.internal.misc
  * @library /test/lib
  * @build sun.hotspot.WhiteBox
- * @run main ClassFileInstaller sun.hotspot.WhiteBox
+ * @run driver ClassFileInstaller sun.hotspot.WhiteBox
  *                              sun.hotspot.WhiteBox$WhiteBoxPermission
  * @run main/othervm -Xbootclasspath/a:. -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI WBApi
  */