Just a small change to make it work wit v2.2 RC1:

find line (68):