Fix to legend issue. fixes #3.