Add code to check that.