Fix shell.py script on tests directory.