fix: add missing types