Fix tests by listening on the correct streams