Change imports to work with django 1.5