Fix  issue with Django shell