A small error fix