fix code compile error