Fix build error from last commit