Small cleanup
Removed leftover code