Fix data bug