Fixed more forgotten imports