Fixed broken lines

Fixed stupid errors in my previous PR #64.

Apologies.