fixed import for simple cases (done)