diff --git a/cli/src/main/java/org/onosproject/cli/net/vnet/VirtualHostCompleter.java b/cli/src/main/java/org/onosproject/cli/net/vnet/VirtualHostCompleter.java
index c44ef40..dec2a35 100644
--- a/cli/src/main/java/org/onosproject/cli/net/vnet/VirtualHostCompleter.java
+++ b/cli/src/main/java/org/onosproject/cli/net/vnet/VirtualHostCompleter.java
@@ -16,7 +16,6 @@
 
 package org.onosproject.cli.net.vnet;
 
-import org.apache.karaf.shell.console.completer.ArgumentCompleter.ArgumentList;
 import org.onosproject.cli.AbstractChoicesCompleter;
 import org.onosproject.incubator.net.virtual.NetworkId;
 import org.onosproject.incubator.net.virtual.VirtualHost;
@@ -37,9 +36,8 @@
 public class VirtualHostCompleter extends AbstractChoicesCompleter {
     @Override
     protected List<String> choices() {
-        ArgumentList args = getArgumentList();
         //parse argument list for network id
-        String[] argsArray = args.getArguments();
+        String[] argsArray = commandLine.getArguments();
         for (String str : argsArray) {
             if (str.matches("[0-9]+")) {
                 long networkId = Long.valueOf(str);