fix for V2.0

Signed-off-by: Edward O'Callaghan <eocallaghan@alterapraxis.com>