Add an extra check which will be used in part 4