Update tests Change-Id: I2821ff19f71d2d7bb4c5022888956e5ff8422306 Signed-off-by: Daniel Haftstein <haftsteind@gmail.com>