Fixed Test Bug