Fix to work with other integrations
0.2.2