Forgot new file for unit tests