Fix build issue introduced by last commit