Updated version, modify just the necessary test