just a smal import fix

Signed-off-by: ClaasJG <claas.j.gramann@gmail.com>