Fix small warning error