Update tests for modified after behaviour