Unit test for this class?