fixed a bug I hope