Trying to fix my screw up