Fix tests for multiple responses