fix compile problem

@kiril self-merging