would be great to have some tests for this class.