Fix tests to handle no response