Bug fix

Add missing header