Fixed build break