Rewrite Widget Test – Shorten