fix missing includes

master has errors right now, fix that