Fix compile error.
Refractor code.