Fix mis-match. Closes #1