Fix incorrect test function signature