Stopped saving attributes and return element on callback instead