Minor update for waffle js

Had a useless `if` in there, other code shortening.