Retain focus hack