Tidy up the interface a bit

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