Correctly fix chained equals assignment statements

Fixes #23