Add fix for cursor bug