Fixed import

I fucked up the first commit because this Github UI is really hard to
understand but... FIXED!