Another bug fix

Signed-off-by: Jack Harley <jackpharley@gmail.com>