small cleanup
fix missing include