add missing 'return' promises in tests