Small test signature fix