Fix for use after free