Updated tests to handle deployment prompting