small cleanup

Still broken