Fix build errors caused by extra argument