Removed not needed code in tests.