Fixing my mistake