Fix a bug I guess