Fix Martyn's code

Signed-off-by: Paul Brook <paul@nowt.org>