Switch to catch unit testing