Add missing import for shell module