Fix4

I added 2 lines
What??