Fixing inputs in mode change