Fixing bug in test.