Fixed little linting and tests error