Add test type fix