updatet unit test to work

For some reason old (not passing) test was committed (sorry...)