Mistake Fixed

Small cleanup