remove unnecessary code from tests