Fix for pagination unit tests

Forgot some classes on previous commit.

Signed-off-by: Dries Vints <dries.vints@gmail.com>