Fix tests due to Rack changes