Fix model

This fixes #37
Did this cause a bug or did you just fix it anyway?