fixed bug I hope