Does your eBike shut-off going up a hill? Can you drive your eBike in sand?
Your problem maybe your small battery and small motor.
Most controllers have a built in minimum voltage level, if your battery sags too much, the controller will shut down.
Also, never use cheap Chinese batteries, they...