Change a bit of logic in buildSrc