Add a better check for enum mode

Add a better check for enum mode #105