Fixed little bug I made