As of now, firejet in the AvatarState is toggle-able. That is canon, but they way the toggle works isn't exactly canon. Here is my suggestion:
When you are in firejet in the avatarstate, instead of working like the real firejet, it should be like how a spout works, using the wasd keys to move...