Moar code fix.