Fixed my tests, I need more specific tests for the new command.