Correct unit test

Signed-off-by: Zack Cerza <zack.cerza@inktank.com>