Prepare for unit testing