Fixed the 'types' issue