Fix my last commit

clean up