Reverted changes to test