Fix to pass linting checks
0.2.2