Oops, fix compile problem