re-add tests