Fix bad previous improvement by me

Sorry, my bad