Make it work without self.response_class too