fix: handle None in init_model