Fix for broken test

Fix #94