Should fix nightly build

The tests addressed here work in a local machine but fail in the nightly build. I think it is due to some parts in the distributed system running very slowly in the nightly build.