Tech Debt: Fixed deprecation warning while running tests