Fixing small mistake