Optimize for single slice case

We don't need to reduce for single slice case. So we don't need to
return all records.

TODO:

  * Clean unit test. We should use real dataset object. Find TODO
    comments.