Small fix for code style