Fixed a bug O.O