small fix with tests