Fix: minor bug fix