adding more unittest coverage