Fix: add missing else