fixed and extended the test