fix missing include (regression of #5)