Fixes for sizes and GMM bug.