fix code again