Fix run logic..