Fix the broken unit test

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