Runtime fix.

- Fixed runtime error from #3298 update.