Test coverage -- fix regression