Fixed build error from last commit