Fix small error