Fix assist registering bug