Add better test, fix with_chef_server