forgot to edit another section of code

for issue #8