Fix: Restore disappeared feature