Updated tests to reflect recent change.