Removing unused lines from test