Fixed some silly bug