Small code style change