Fixes AAM Advanced notebook

Fixes the `AAM Advanced` notebook