Code cleanup

Ah I see, will fix.