fixed small compile problem