Yea, I was thinking of that too, but implementing it is too challenging for me so I opted for a rough fix :)
