Fix build break caused by b608cfe