Fix Server.run tests