fix compile error, take 2