Fix imports for Shockwave