Resolved an issue in tests