Fixed issue with stratification