add tests which was removed