Shimano BT-EN 805L Replacement

[Bish1987]

Hi, I have a Marin rift Zone E with the 805L 504watt battery.

Could anyone confirm if I could replace this with BT-EN 8036 600watt. It has the EP6 Motor.

TIA