Fixing a test (#67)