remove unused code from test