Try to fix issue in test