Fix non passing test (add mockPrompt calls)