Fix create button bug