Fix types in a test

Change-Id: I2f857f564e100cc4cf6b885c5a8f96e506b4046f