Fix build break