Adding the corresponding tests