Fixed build error of last commit