Helper class for tests