Address comment in #50

Fixes #50