Fixed Test assertion.