fix code warning