Script: Fix imports