add missing includes

Signed-off-by: Joshua Rubin <jrubin@zvelo.com>