Small fix applied