Change equals of enum to ==