Fixed incorrect back port and missing import

Signed-off-by: Stefan Marr <git@stefan-marr.de>