Fix: minor bug