Fix some missing function flow