Fix missing imports for scripts without .py suffix