Fix get_all method and set_model