fix remaining failing test