fixing bug that caused return sets of 1 to not work as expected