Fix key error in clean