Fix

Fix the wrong name of model