Fix up small thing in test