Fixing broken unit test after rebase

Signed-off-by: Don Kjer <don.kjer@gmail.com>