Fixed a minor bug by removing parts of old code.