Fixed bug I made