fix remaining main test