Fixed the related test.