small fix after tests.