Fix some function mismatch