fix code works