forgot to move the test class in line with the real one.. oops!