Add unit directory to test