Fix warning/build