Fix run method bug