Add tests for different returns of proc