fixed the bug I did