Lint the test code

This should fix the `travis-ci` build.