Updated tests for new command