Made it compatible with Battlemod 2