Update unit tests to reflect renamed function

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