fix bug, I think