Fixed broken unit test. This resolves issue #25