fixing my code

shame on me