Fix model

This fixes #37
Just a fix.