fix tests for 4.1