Fix some missing imports and a zero covariance matrix.