Forgot to add new test class