fixing ama example notebook