added more checks for ensemble collapse