Fixed warning in test