Put wait.for back in, I realised how important it is

This reverts commit 706f5d8a6ebbda36ea289e58fdc1dc48b3aec963.

Conflicts:
	package.json