Fixed omitted import in output parser tests