fixed a bug with base models