Quick fix

Signed-off-by: Sam Hellawell <sshellawell@gmail.com>