Small fix #244