Correct a bug with display property