minor fix code