Updated to new model variable

Fixes #19