Minor Bug Fix

Ready for Integration with Wi-Net