Update tests for last change