Update tests for last change (2)