Issue #8 fix

Is mainly cosmetic so no hurry to update.