Wheel module fixes