fixed tests

Signed-off-by: Darren Schwarz <darrenschwarz@gmail.com>