diff --git a/tests/test-amp-img-sanitizer.php b/tests/test-amp-img-sanitizer.php
index 14dfe8e4cc7..e77274c1457 100644
--- a/tests/test-amp-img-sanitizer.php
+++ b/tests/test-amp-img-sanitizer.php
@@ -214,6 +214,16 @@ public function get_data() {
'',
null,
),
+
+ 'img_with_sizes_attribute_removed' => array(
+ '',
+ '',
+ ),
+
+ 'amp_img_with_sizes_attribute_retained' => array(
+ '',
+ null,
+ ),
);
}