Small fix to test.

Signed-off-by: Vic <vwalldev@gmail.com>