Couple edits to make it work in zepto