load_model! was in the wrong place