Fixing missing tests error