Should be changed in `register` widget as well