Fixing wrong fit example

Fixes #2