Setup for unit testing