Fixing a bug; should add tests