Fixing tests after merge

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