IDEA 2016.3 + Gradle - Run Before Rrebuild fails.

I'm using IntelliJ IDEA for a Gradle project. I have a custom task for generating source code, which I've checked for running 'Before Rebuild'. Everything worked fine until recent update, now my task isn't being called. Am I missing something, or is this a bug?