Lego train wheelset slide/rotate top

Lego train wheelset slide top. Not 100% accurate but it fits. My kid snapped the original rotational axis after crashing the train. So this is my attempt to fix it. In my case parts joined on top do spin but don't spin freely. It comes down to your printer settings. Some sanding required because of the sloped top surface.

Printed as PLA with 0.2 extruder and 0.2 layer height.

Based on existing Lego brick generator, so that why the bottom part fits quite well.