Fixed import

Signed-off-by: Bjoern Weidlich <bjoernweidlich@gmail.com>