Add missing input entity