small code fix
cleaned up include a bit