Adding some sleep to fix flakey test