Updated tests for previous commit