Documentation isn't building on RTD

Move some module imports around