Bugfix in unitary