I think you should explicitly import classes