Focus is now properly working without a delay