Fixed minor bug