fix/add tests