fix broken test fixes #225