Return value fix

Signed-off-by: Mike Griffiths