Fixed minor bug..