Tech Debt: Removed mock on test