Remove code specific for the test environment