Fix small bug for successful run