Minor update to wait-for-container script