Fixing null/nil bug.