fixed a bug I guess