fixed compile error. haven't test yet