Fixed issue with types button after import