fixes slight error in tests