Fix up it tests