Fixing issue in test.