Fix issue with not prompting output