Fix error when running with --irb option