Removing sleep from testing