fixed model type
Merge pull request #14 from Charlespernet/fixing_models

fixed model type