Fixing unittest