Fix small mistake