Fix test linting error

(my bad)