bug fix for non composite key scenario