remove unnecessary method from test code