fix missing tests