Fixed prompt issue