legacy: import fix

Signed-off-by: Jiri Kuncar <jiri.kuncar@cern.ch>