add tests fix failure