Fixed the test by inheriting CLITestCase