Minor change to roll back code to what it looked before.