Tiny patch

Fixes one problem. Just like I said, if you have any problems, post a comment. I will fix them as fastest I can.