Remove code for unit tests.