fix remaining test