Fix to pass linting checks
0.4.1