Improve test_cmd