Fixed Database

Fixed two models