Small fix

Remove unnecessary code