Removing lines of test code