fixed a little bug. I hope