general: importing modules from packages fix

Signed-off-by: Eirini Psallida <eirini.psallida@cern.ch>
Can you join it with next line to make it nicer?