I feel that we need some unit test on this class.