fixed a bug with the wrong imported class