Handle null values
Bug fix