Fixed a bug..