Update tests for previous commit