Updated to fix config