Adjust the logic of one method in Action class