FIX: minor bug