Fix --test build