Fixed revert in last commit

Code was added, not replaced.