[hotfix] Fix compilation error

The includes were wrong. We probably have to include the whole <memory>