Fix async bug