Minor tweak to code

hi zach