Fix for #104

Fixes an issue where count queries were going to the wrong shard