Fixes minor issue with PR #121.