Small fix in complexity