You should unit test this class