Clear code - Small FIX

Now the code works as it should.