Bug fixed for test.

Hot fix for test.