Fixed import for newer Django versions