Fixed minor bug with the REPL