fix missing include in last commit