Fix: add BehaviorType import