fixed bug I just made