What about tests for this class?