Another minor tweak. Fixes #9