Re-added accidentally removed line of code