Fix minor bug introduced in the last commit.
Merge pull request #33 from materialsvirtuallab/master

Fix minor bug introduced in the last commit.