Add missing module import

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