Fix bug when loading deployment model