fixing minor bug in previous commit