This class is only used in tests. Move to test source tree?