fix pubsub tests.

Fixing the regression tests from the #330 discoveries.