add tests fix