small bug fix for compile error