Patch a simple unittest problem