fixed some broken imports