yay, fixed a bug