This is now fixed. It was pretty easy to fix.
Credit to @CheetahAnimae for the original code, modified by me.
