adapted tests to code change