Small fix in Solution