Add checks for more versions of django