Fixed feedback from linter