fix code (2)