Fix for submit button not working