updated test. doesn't work in hadoop mode actually