Small code style fix