A little changes to DBSet