fixing minor bug