Fix var assignment in test