[fix][amg] Add missing include See merge request !2738 (cherry picked from commit 0c8d203e) 707dc14b [fix][amg] Add missing include