fix bug, I hope