fix 2 lines issue