Removed extra stuff from test