Fix unit tests.

Signed-off-by: David Scott <dave.scott@eu.citrix.com>