Small fix on latest update