Added some code to try to fix controls.