Fix publish tests