Pass the test

This should fix the test, as discussed in #14