Fix error in test.
Fix exception in test.