Good catch. That was a copy pasta from my ipython notebook while I was testing it :) Changing it to their own lines.