Fixed a bug I made