FIX: minor bug fix