Fix label bug

Closes issue #1 (I thought issues and pull requests were integrated somehow, but can't find the option)