fix for diversity test