Fixing a minor bug (#31)