Change test for change to publish

Due to the change to the publish method the tests had to be updated.