Attempting to fix previously mentioned error