Fix compile error

Fix compile error