Little fix

I think it's gonna be useful. :)