Amend last commit, forgot one test class