Fix build error

Add missing include file.