Add missing types in trait