Small code styling tweak