Small fix update