Fix Bug in `test_shift`

Fixed a minor bug, corresponding to value of `n` returned in
the case with 2 members in the add dict