Fix swap model for set size 2. Fixes #22.

Explicity say which dimension to sum on, in case we only have 1
distractor/trial.