Small fix #5