Add validation on response type