fixed the test that was broken after latest changes