fix to unit testing