Fix up test issue