Move check_os call to prepare_env.sh