Fixed the response class