Add the missing tests