[TASK] Uncomment more tests