Changed IF stage so multiple drivers issue should not occur