Fix related test.