Fix model

This fixes #37