Fixed minor issue with prompt