Fixed missing setup import for TestCommand