Add missing import

The class is used in theme:body codes.