Fix failing webdriver test due to change in Module edit UI