Electric motor seems to be a tight fit

I have changed the bearings in this hub however when i put the 2 parts back together they seem like a tight fit what if anything can be done to make this work properly when useing the bike it heats up and can only ride it half a mile because it is a tight fit thanks for any kind of help.anyone out there ever resolving this issue would be a great help thanks again.