Simple cleanup

Works on my machine!