Fixing build break