Fix bug: some optimisers return theta as a column vector