Fix ref counter bug