Nokia on Friday said it has a fix for data connection problems that plagued the recently launched Nokia Lumia 900.

The company acknowledged a memory management issue was discovered that could, in some cases, lead to loss of data connectivity.

“This issue is purely in the phone software, and is not related to either phone hardware or the network itself,” Nokia wrote on its Nokia Conversations blog.

Those who have already purchased a Lumia 900 will have one of two options. Those users can either update the device with the latest software via Zune “on or around Monday, April 16th,” or they can swap their unit for an updated Lumia 900.

Nokia had already announced that everyone who has purchased a Nokia Lumia 900, or who will purchase one between now and April 21st, will receive a $100 credit to their AT&T bill from Nokia.

The $100 credit makes the phone essentially free on AT&T with a two-year contract. AT&T is currently offering the Lumia 900 for $99. A report from IHS iSuppli released last week estimates that the bill of materials (BOM) on the Lumia 900 is $209.