fix for main script v. 0.1.0