Fix regression bug from e2d32b7fba9538ba1a18a80a97f45ff8679b26a1

Fixes #4.