small fix to check for none