Adjust test. Closes #9