Fixing minor errors in tests