Fix 'python -i' mode