Fix test

This is related to #21.

CC:  @fbergroth