Infinispan Server CLI container command doesn't work

The Infinispan Server CLI has a "container" command which doesn't appear to work anymore under 8.0.1. I suspect it's because the Infinispan subsystem was renamed from "infinispan" to "datagrid-infinispan", however I can't find where that change should be made! It appears it should be in the CLI Connection implementation, but I can't for the life of me find the HTTP Connection that I believe is being used in this case.

If someone can point me to the right class I'd be happy to change and test the fix.