legacy: fix import problems

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