Trying to fix my mistake