Fix test. Closes #106