Minor fix to REPL