Fixed Test Driver. Kept saying test failed even though it suceeded