Fix breakage introduced by #21
Merge pull request #25 from square/edenman/fix-set-selected

Fix breakage introduced by #21