Adds None check