Updated tests for plugin_for_command to reflect new behaviour