Fixed problems after linting