refactored weapon aftereffects test a little