Move this to the setup, remove the sleep