More fixes to the REPL dialog