Small fix in code format