Fix oops in test