Fix small warning