Adapt for testing environment