Fixed minor bug...