fix(shake): stop using Omit on return type and give filter parameter a safer type

fix(shake): stop using Omit on return type and give filter parameter a safer type #519