fix code overflow