Out of bounds rules clarification

[joebjay20]

I have a question about a situation that arose in a recent pickup game. Player A1 misses a layup that is contested but not blocked by player B1. Player A1 misses the layup but steps out of bounds before recovering to rebound and hit the second attempt. I was always under the impression that if you were the last person to touch the ball and you went out of bounds, you could not be the first to touch the ball once establishing yourself back inbounds, but the opposing players argued that this provision did not apply if the player in question attempted a shot. I have scanned rule books but could not find anything relating to this particular situation, any insight anyone could provide is greatly appreciated.