modification for tests