Changed code to match other linters