Fixes #111

Minor change