Fix to match @jasondegraw version