Updated to work with Statamic 1.6.7