Add missing attribute to flask storage