Fixes build break