Fix test. Closes #179