bug fix. null check