Adds back in test, but passing this time.