fix problem with test