[PATCH] Add missing includes to headers