Update test

Fixes #185