Hello kb Welcome to the biggest and best Motorized Bicycle forum.
Check and make sure you alignment is correct and you have a 1/2 to 3/4" inch slack on the chain.
If you have the dished sprocket sometimes you have to flip it to get the correct alignment. Also make sure there is no wobble in the rear sprocket.