[TASK] Removed unused code from tests