Fix missing tests