fixed two failed tests