Fixed issue with design