small fix code