Code cleanup

True, will fix that.