minor bug fixed from previous commit