Removing unused lines in tests