Fix book association and back buttons