Failed the second stage as my intuition was correct but did not pass all the tests. It took me 5 minutes to fix it afterward and was confident that the interview went well given just a minor mistake.