@AfterTest public void cleanUp() { if (command != null) { command.getClient().closeConnection(); } if (cubeCommands != null) { cubeCommands.getClient().closeConnection(); } }