Added missing import

Signed-off-by: Dominik Schadow <dominikschadow@googlemail.com>