Fixed a minor bug that came with previous commit

ed104f5