Fixes

The last update was nearly working but I realized it was bugging severely after publishing it sooooo....yeah....

Here's a fix ^^'