Fixing expectation in diamond gradle test