Fix tests for set_token command

includes nice error throwing when command not specified properly