Trying to fix run loop issue