Fixed big bug!!!

Hi. I'm not sure how it worked before, but with my edit - it works great :)