Fixed error on diagram