Fixed minor issue in test