Fix: type and proper header