Simplify tests

Change-Id: Ide435d4d11e9ab03c953ed91e949d55e46c5137e