Small fix #3