Fix for last commit

Signed-off-by: Nick Brook <nick@rmpenterprise.co.uk>