fix tests for new behaviour