Fixes for benchmark package

Changes in fitting syntax, only tested the AAM