Fix script - add modules