Fixed some issues with the controls.