Fixes #8

One more code change.