Fixed build error on last commit