fix compilation issue. #2