Fix publishing unit tests