Refinery21 compat

Update extension to be compatible with Refinery 2.1.