fix for unit tests

Signed-off-by: Evan Vetere <ecv@sine.com>