Little fix with signature to get publish tests pass