Fixed small error

Small fix