fixed compilation eror due to refactoring