Fix up Lettuce tests