Fix bug.

Bug due to code migration.
1 file changed