minor bug squashing
minor bug squashing