Fixed my bad coding