[PATCH 2/2] add some missing includes