Fixes issue #124 -- required small modification to code to implement