add correct tearDown in stage