Yeah, it's obvious that this is going to break as we add more unit tests. In this case, we'll need another way of associating the test with the class that's being tested (The java generated tests currently require knowledge about the class under test)