[Unit Testing] Fixed broken expectations.